import numpy as np

a = np.array([[3, 7], [9, 1]])
print(a)
print(np.sort(a))

a = np.array([3, 1, 2])
print(a)
print(np.argsort(a))
